package com.tjetc.config;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

@Configuration
public class ThreadPoolConfig {
    
    @Bean("aiThreadPool")
    public ExecutorService aiThreadPool() {
        return new ThreadPoolExecutor(
            5,  // 核心线程数：保持5个线程随时可用
            10, // 最大线程数：最多创建10个线程
            60L, // 空闲时间：空闲线程60秒后销毁
            TimeUnit.SECONDS, // 时间单位
            new LinkedBlockingQueue<>(100), // 队列：最多排队100个任务
            new ThreadPoolExecutor.CallerRunsPolicy() // 拒绝策略：队列满时让调用线程自己执行
        );
    }
    
    @Bean("messageThreadPool")
    public ExecutorService messageThreadPool() {
        return new ThreadPoolExecutor(
            10,  // 核心线程数
            20,  // 最大线程数
            60L, // 空闲时间
            TimeUnit.SECONDS,
            new LinkedBlockingQueue<>(200), // 队列容量更大
            new ThreadPoolExecutor.CallerRunsPolicy()
        );
    }
    
    @Bean("dbThreadPool")
    public ExecutorService dbThreadPool() {
        return new ThreadPoolExecutor(
            5,   // 核心线程数
            15,  // 最大线程数
            60L, // 空闲时间
            TimeUnit.SECONDS,
            new LinkedBlockingQueue<>(500), // 数据库操作队列更大
            new ThreadPoolExecutor.CallerRunsPolicy()
        );
    }
}